"""
./test/ ---------> /home/demo/Desktop/test/

思路：
1、定义变量，保存源文件夹、目标文件夹所在的路径
2、在目标路径创建新的文件夹
3、获取源文件夹中的所有的文件（列表）
4、遍历列表，得到所有的文件名
5、定义函数，进行文件拷贝

文件拷贝函数：
参数： 源文件夹路径 目标文件夹路径 文件名
1、拼接源文件和目标文件的具体路径
2、打开源文件，创建目标文件
3、读取源文件的内容，写入到目标文件中 （while）

"""
import os 
import multiprocessing
def copy_work(source_dir, dest_dir, file_name):
    # print(multiprocessing.current_process())
    # print("dest_dir-->",dest_dir)
    # print("source_dir-->",source_dir)
    # print("file_name-->",file_name)
    source_path = source_dir + '/' + file_name
    dest_path = dest_dir + "/" +file_name
    print("source_path-->",source_path)
    print("dest_path-->",dest_path)   

    # 3. 读取源文件的内容,写入到目标文件中(while)
    with open(source_path,'rb') as source_file:
        # 创建目标文件
        with open(dest_path,'wb' ) as dest_file:
            while True:
                # 读取 源文件, 保存到目标文件
                file_data = source_file.read(1024)
                # 判读文件是否读取完成
                if file_data:
                    dest_file.write(file_data)
                else:
                    break





if __name__ == '__main__':
    # 1.定义变量,保存源文件夹,目标文件夹所在的路径
    source_dir = "./test"
    dest_dir = "/usr/python_leaning/process_dir/dest_test"
    # 2.在目标路径创建新出的文件夹
    try:
        os.mkdir(dest_dir)
    except Exception as e:
        print('文件夹已存在了')
    
    # 3.获取源文件夹中 的所有的文件(列表)
    file_list = os.listdir(source_dir)

    # 创建进程池
    pool = multiprocessing.Pool(3)

    # 4.遍历列表 得到所有文件的文件名
    for file_name in file_list:
        # print(file_name)
        # 5. 定义函数, 进行拷贝
        pool.apply_async(copy_work,(source_dir,dest_dir,file_name))
    
    pool.close()
    pool.join()
